#!/usr/bin/env python
# -*- conding: utf-8 -*-

"""
@Time     : 2024/8/31 21:42
@Author   : liujingmao
@File     : 03.百度千帆文本嵌入模型.py
"""

# 使用国内的大模型，返回速度比较快
import dotenv
from langchain_community.embeddings.baidu_qianfan_endpoint import QianfanEmbeddingsEndpoint

dotenv.load_dotenv()

embeddings = QianfanEmbeddingsEndpoint()

query_vector = embeddings.embed_query("我叫慕小课，我喜欢打篮球游泳")

print(query_vector)
print(len(query_vector))
